import pandas as pd


# 读取文件中的数据
def read_txt(path):
    """
    读取.txt文件，将文件中的内容以对象的形式返回
    :param path: 文件路径
    :return: {
        ...,
        "zrp": [],
        "time": []
    }
    """
    with open(path, 'r', encoding='utf-8') as file:
        arr = []
        # 逐行读取文件
        for line in file:
            arr.append(line.split())
        obj = {}
        for key in arr[0]:  # 先读取第一行，将第一行中每一列的值作为属性名
            obj[key] = []
        for i in range(1, len(arr)):  # 将剩下行的每一列的值，放入到obj中对应属性的的数组中
            for j in range(len(arr[i])):
                obj[arr[0][j]].append(arr[i][j])
    return obj


# 保存数据到文件
# data为普通对象
# path为保存的文件路径
def save_txt(data, path):
    fileData = pd.DataFrame(data)
    fileData.to_csv(
        path,
        sep='\t',
        index=False
    )
